package util;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.HorizontalAlignment;

import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author <a href="mailto:hellohesir@gmail.com">Mr_He</a>
 * 2019/10/16 16:02
 */
public class ExcelUtil {

    public static void main(String[] args) throws IOException {
        List<String> data1 = new ArrayList<String>();
        data1.add("a");
        data1.add("b");
        data1.add("c");
        data1.add("d");
        data1.add("e");
        List<List<String>> list = new ArrayList<>();
        list.add(data1);
        list.add(new ArrayList<>(data1));
        createExcel(list, new String[]{"a1", "b2", "c3"});
    }

    public static void createExcel(List<List<String>> dataList, String[] titleArray) {
        // 第一步，创建一个webbook，对应一个Excel文件
        HSSFWorkbook wb = new HSSFWorkbook();
        // 第二步，在webbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet("sheet1");
        sheet.setDefaultColumnWidth(20);// 默认列宽
        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
        HSSFRow row = sheet.createRow((int) 0);
        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        // 创建一个居中格式
        style.setAlignment(HorizontalAlignment.CENTER);

        // 添加excel title
        HSSFCell cell;
        for (int i = 0; i < titleArray.length; i++) {
            cell = row.createCell((short) i);
            cell.setCellValue(titleArray[i]);
            cell.setCellStyle(style);
        }

        // 第五步，写入实体数据 实际应用中这些数据从数据库得到,list中字符串的顺序必须和数组strArray中的顺序一致
        int i = 0;
        for (List<String> list : dataList) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < titleArray.length; j++) {
                String value = "";
                if (list.size() >= j + 1) {
                    value = list.get(j);
                }
                row.createCell((short) j).setCellValue(value);
            }

            // 第六步，将文件存到指定位置
            try {
                FileOutputStream fout = new FileOutputStream("D:\\project\\climb\\test.xls");
                wb.write(fout);
                fout.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
            i++;
        }
    }

}
